package xyz.sekai52799.answer.a2001_a2200;

import java.util.*;
import java.util.function.Function;
import java.util.stream.Collectors;

public class Question2007 {
//    一个整数数组 original 可以转变成一个 双倍 数组 changed ，转变方式为将 original 中每个元素 值乘以 2 加入数组中，然后将所有元素 随机打乱 。
//
//    给你一个数组 changed ，如果 change 是 双倍 数组，那么请你返回 original数组，否则请返回空数组。original 的元素可以以 任意 顺序返回。
//
//
//
//    示例 1：
//
//    输入：changed = [1,3,4,2,6,8]
//    输出：[1,3,4]
//    解释：一个可能的 original 数组为 [1,3,4] :
//            - 将 1 乘以 2 ，得到 1 * 2 = 2 。
//            - 将 3 乘以 2 ，得到 3 * 2 = 6 。
//            - 将 4 乘以 2 ，得到 4 * 2 = 8 。
//    其他可能的原数组方案为 [4,3,1] 或者 [3,1,4] 。
//    示例 2：
//
//    输入：changed = [6,3,0,1]
//    输出：[]
//    解释：changed 不是一个双倍数组。
//    示例 3：
//
//    输入：changed = [1]
//    输出：[]
//    解释：changed 不是一个双倍数组。
//
//
//    提示：
//
//            1 <= changed.length <= 105
//            0 <= changed[i] <= 105

    // 排序
    public int[] findOriginalArray(int[] changed) {
        if (changed.length % 2 == 1){
            return new int[0];
        }
        int[] result = new int[changed.length / 2];
        Arrays.sort(changed);
        Map<Integer, Long> count = Arrays.stream(changed).boxed().collect(Collectors.groupingBy(Function.identity(), Collectors.counting()));
        int i = 0;
        for (int a : changed){
            if (count.get(a) == 0){
                continue;
            }
            count.put(a,count.get(a) - 1L);
            if (count.getOrDefault(a*2, 0L) == 0L){
                return new int[0];
            }
            count.put(a*2, count.get(a*2) - 1L);
            result[i] = a;
            i++;
        }
        return result;
    }
}
